from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from langchain_community.llms import Ollama

# 初始化 Ollama 模型
ollama_llm = Ollama(model="deepseek-r1:8b")
# 创建提示模板
prompt_template = "请为生产{product}的公司起一个名字。"

# 构建 LLMChain
llm_chain = LLMChain(
    llm=ollama_llm,
    prompt=PromptTemplate.from_template(prompt_template)
)

# 运行链并获取结果
result = llm_chain.run("彩色袜子")
print(result)
